package com.example.zsgc.dao;


import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.sql.DataSource;
import java.io.IOException;
import java.io.InputStream;
import java.sql.*;
import java.util.Properties;


/**
 * 执行数据库操作的工具类。
 */
public class BaseDao {
    private static String driver =null;// 数据库驱动字符串
    private static String url = null;// 连接URL字符串
    private static String username =null; // 数据库用户名
    private static String password =null; // 用户密码

    protected Connection conn;

    static {
        init();
    }

    public static void init(){
        Properties params=new Properties();
        String configFile="database.properties";

        try {
            //加载配置文件到输入流
            //获取当前类的加载器对象
            InputStream is=BaseDao.class.getClassLoader().getResourceAsStream(configFile);
            //从流中读取键值对
            params.load(is);
            //通过key读取Value
            driver=params.getProperty("driver");
            url=params.getProperty("url");
            username=params.getProperty("username");
            password=params.getProperty("password");
            //获取完毕以后可以加载驱动类
            Class.forName(driver);
        } catch (IOException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }

    }



    /**
     * 获取数据库连接对象。
     */
    public Connection getConnection() {
        // 获取连接并捕获异常
        try {

            conn = DriverManager.getConnection(url, username, password);
        } catch (SQLException e) {
            e.printStackTrace();
        }
        return conn;// 返回连接对象
    }

    /**
     * 通过数据源获取数据库连接对象。
     */
    public Connection getConnection2() {
        // 获取连接并捕获异常
        Context ctx;
        try {
            ctx = new InitialContext();
            // 通过逻辑资源名获取数据源对象
            DataSource ds = (DataSource) ctx.lookup("java:comp/env/jdbc/onlineauctionsystem");
            // 从数据源中获取连接对象
            conn = ds.getConnection();
        }  catch (NamingException e) {
            e.printStackTrace();
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
        return conn;// 返回连接对象
    }

    /**
     * 关闭数据库连接。
     *
     * @param conn
     *            数据库连接
     * @param stmt
     *            Statement对象
     * @param rs
     *            结果集
     */
    public void closeAll(Connection conn, Statement stmt, ResultSet rs) {
        // 若结果集对象不为空，则关闭
        try {
            if (rs != null && !rs.isClosed())
                rs.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
        // 若Statement对象不为空，则关闭
        try {
            if (stmt != null && !stmt.isClosed())
                stmt.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
        // 若数据库连接对象不为空，则关闭
        try {
            if (conn != null && !conn.isClosed())
                conn.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 增、删、改操作
     *
     * @param sql
     *            sql语句
     * @return 执行结果
     */
    protected int executeUpdate(String sql, Object... params) {
        int result = 0;
        conn = this.getConnection();
        PreparedStatement pstmt = null;
        try {
            pstmt = conn.prepareStatement(sql);
            for (int i = 0; i < params.length; i++) {
                pstmt.setObject(i + 1, params[i]);
            }
            result = pstmt.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            closeAll(null, pstmt, null);
        }
        return result;
    }

    /**
     * 查询操作
     *
     * @param sql
     *            sql语句
     * @param params
     *            参数数组
     * @return 查询结果集
     */
    protected ResultSet executeQuery(String sql, Object... params) {
        conn = this.getConnection();
        PreparedStatement pstmt = null;
        ResultSet rs = null;
        try {
            pstmt = conn.prepareStatement(sql);
            for (int i = 0; i < params.length; i++) {
                pstmt.setObject(i + 1, params[i]);
            }
            rs = pstmt.executeQuery();
        } catch (SQLException e) {
            e.printStackTrace();
        }finally {
            closeAll(conn,pstmt,rs);
        }
        return rs;
    }
}